#ifndef INFERENCE_H
#define INFERENCE_H

// Cpp native
#include <string>
#include <vector>
#include <map>

class BertInfer
{
public:

    static BertInfer* getInstance();
    std::map<std::string, std::vector<std::string>> runInference(const std::string &input);
private:
    static BertInfer* instance;  // Pointer to the single instance of the class
    BertInfer(const std::string &modelPath, const std::string &vocabPath);
    // Private constructor so that no objects can be created.

};

#endif // INFERENCE_H
